package com.fly.lambda.train;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class LambdaTest001
{
    private static List<Transaction> transactions = new ArrayList<>();
    
    @BeforeAll
    public static void init()
    {
        log.info("★★★★★★ @BeforeAll init ★★★★★★");
        Trader raoul = new Trader("Raoul", "Cambridge");
        Trader mario = new Trader("Mario", "Milan");
        Trader alan = new Trader("Alan", "Cambridge");
        Trader brian = new Trader("Brian", "Cambridge");
        
        transactions.add(new Transaction(brian, 2011, 300));
        transactions.add(new Transaction(raoul, 2012, 1000));
        transactions.add(new Transaction(raoul, 2011, 400));
        transactions.add(new Transaction(mario, 2012, 710));
        transactions.add(new Transaction(mario, 2012, 700));
        transactions.add(new Transaction(alan, 2012, 950));
    }
    
    // (1) 找出2011年发生的所有交易，并按交易额排序（从低到高）
    // (2) 交易员都在哪些不同的城市工作过？
    // (3) 查找所有来自于剑桥的交易员，并按姓名排序
    // (4) 返回所有交易员的姓名字符串，按字母顺序排序
    // (5) 有没有交易员是在米兰工作的？
    // (6) 打印生活在剑桥的交易员的所有交易额之和
    // (7) 所有交易中，最高的交易额是多少？
    // (8) 找到交易额最小的交易
    @Test
    public void test()
    {
    }
    
    @Test
    public void test01()
    {
        log.info("找出2011年发生的所有交易，并按交易额排序（从低到高）");
        transactions.stream()
            .filter(t -> t.getYear() == 2011) // 2011年
            .sorted(Comparator.comparing(Transaction::getValue))// 排序
            .forEach(t -> log.info("{}", t));
        
        transactions.stream()
            .filter(t -> t.getYear() == 2011)// 2011年
            .sorted(Comparator.comparingInt(Transaction::getValue)) // 排序
            .forEach(t -> log.info("{}", t));
    }
    
    @Test
    public void test02()
    {
        log.info("交易员都在哪些不同的城市工作过？");
        transactions.stream()
            .map(t -> t.getTrader().getCity()) // 城市
            .distinct() // 去重
            .forEach(log::info);
        
        transactions.stream()
            .map(t -> t.getTrader().getCity()) // 城市
            .distinct() // 去重
            .collect(Collectors.toList())// 收集
            .forEach(log::info);
    }
    
    @Test
    public void test03()
    {
        log.info("查找所有来自于剑桥的交易员，并按姓名排序");
        transactions.stream()
            .filter(t -> t.getTrader().getCity().equals("Cambridge"))// 剑桥
            .map(Transaction::getTrader) // 交易员
            .distinct() // 去重
            .sorted(Comparator.comparing(Trader::getName)) // 排序
            .forEach(t -> log.info("{}", t));
        
        transactions.stream()
            .filter(t -> t.getTrader().getCity().equals("Cambridge"))// 剑桥
            .map(Transaction::getTrader) // 交易员
            .distinct() // 去重
            .sorted(Comparator.comparing(Trader::getName)) // 排序
            .collect(Collectors.toList())// 收集
            .forEach(t -> log.info("{}", t));
    }
    
    @Test
    public void test04()
    {
        log.info("返回所有交易员的姓名字符串，按字母顺序排序");
        transactions.stream()
            .map(t -> t.getTrader().getName()) // 交易员名字
            .distinct() // 去重
            .sorted() // 排序
            .forEach(log::info);
        
        transactions.stream()
            .map(Transaction::getTrader) // 交易
            .collect(Collectors.toList())// 收集
            .stream()
            .map(Trader::getName) // 交易员名字
            .distinct() // 去重
            .sorted() // 排序
            .forEach(log::info);
    }
    
    @Test
    public void test05()
    {
        log.info("有没有交易员是在米兰工作的？");
        log.info("{}", transactions.stream().anyMatch(t -> t.getTrader().getCity().equals("Milan")));
    }
    
    @Test
    public void test06()
    {
        log.info("打印生活在剑桥的交易员的所有交易额之和");
        int total = transactions.stream()
            .filter(t -> t.getTrader().getCity().equals("Cambridge")) // 剑桥
            .mapToInt(Transaction::getValue) // 交易额
            .sum();
        log.info("{}", total);
    }
    
    @Test
    public void test07()
    {
        log.info("有交易中，最高的交易额是多少？");
        int max = transactions.stream()
            .mapToInt(Transaction::getValue) // 取交易额
            .max() // 最高
            .getAsInt();
        log.info("{}", max);
        
        int max2 = transactions.stream()
            .max(Comparator.comparingInt(Transaction::getValue)) // 最高交易额
            .get() // 获取对象
            .getValue();
        log.info("{}", max2);
    }
    
    @Test
    public void test08()
    {
        log.info("找到交易额最小的交易");
        Transaction min = transactions.stream().min(Comparator.comparing(Transaction::getValue)).get();
        log.info("{}", min);
    }
}

/**
 * 交易员
 */
@Data
class Trader
{
    private final String name;
    
    private final String city;
    
    /**
     * @param name 姓名
     * @param city 城市
     */
    public Trader(String name, String city)
    {
        this.name = name;
        this.city = city;
    }
}

/**
 * 交易
 */
@Data
class Transaction
{
    private final Trader trader;
    
    private final int year;
    
    private final int value;
    
    /**
     * @param trader 交易员
     * @param year 年份
     * @param value 交易额
     */
    public Transaction(Trader trader, int year, int value)
    {
        this.trader = trader;
        this.year = year;
        this.value = value;
    }
}
